Absolute value in Python numpy